import json
import os
import sqlite3

from utils.db_utils import db_new_connection

def create_bilibili_main_user_db():
    # 连接 SQLite 数据库
    conn = db_new_connection()
    cursor = conn.cursor()

    # 检查表是否存在，如果不存在则创建
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS mainUser (
        pkId INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT,
        mid TEXT,
        avatar TEXT
    )
    ''')

    print("main表创建成功")
    return True

def create_bilibili_user_db():
    # 连接 SQLite 数据库
    conn = db_new_connection()
    cursor = conn.cursor()

    # 检查表是否存在，如果不存在则创建
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS biLiUser (
        pkId INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT,
        talkerId TEXT,
        avatar TEXT,
        mid TEXT
    )
    ''')

    print("bilibili表创建成功")
    return True

def create_bilibili_direct_message_db():
    # 连接 SQLite 数据库
    conn = db_new_connection()
    cursor = conn.cursor()

    # 检查表是否存在，如果不存在则创建
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS biLiMessage (
        pkId INTEGER PRIMARY KEY AUTOINCREMENT,
        mid TEXT,
        senderUid TEXT,
        receiverId TEXT,
        content TEXT,
        msgSeqno TEXT,
        timestamp TEXT,
        msgType TEXT,
        msgKey TEXT,
        msgStatus TEXT,
        msgSource TEXT
    )
    ''')

    print("私聊表创建成功")
    return True